dictionary comprehension with if else